# pythonProject5/schemas/record.py
from pydantic import BaseModel
from typing import Optional
from datetime import datetime


class ASRRecordCreate(BaseModel):
    filename: str
    text: str
    language: str
    duration_seconds: float
    file_size: int
    audio_path: Optional[str] = None
    model_used: Optional[str] = "whisper-small"


class TranslationRecordCreate(BaseModel):
    source_text: str
    target_text: str
    source_language: str
    target_language: str
    duration_seconds: float
    source_audio_path: Optional[str] = None
    model_used: Optional[str] = "whisper-small"
    translation_engine: Optional[str] = "nllb"
